Closes #316

✅ Refactoring Complete

I've successfully refactored the WebSocket client to eliminate global scope pollution and enable advanced connection management.

Key Improvements:

Encapsulated Architecture:

• TraceStreamClient class wraps all socket initialization and event listener logic
• Connection state (nodeCount, startTime, ws reference) now lives within instance, not globally
• All 4 event listeners (open, message, error, close) are bound cleanly inside attachEventListeners()

Graceful Connection Management:

• connect() method returns a promise, allowing proper async handling
• cleanup() method removes all listeners and nullifies the socket reference—enabling garbage collection of dead sockets
• close() method gracefully closes connections before attempting reconnection
• State flags (isConnecting, isClosed) prevent duplicate connections

Factory Pattern:

• connectToTraceStream() factory function provides a clean API for creating and connecting clients
• Can easily instantiate new clients for reconnection without polluting global scope

Organized Message Handling:

• handleMessageType() routes messages to dedicated handler methods
• Each message type (trace_started, trace_node, etc.) has its own handler

This architecture now supports:

• ✅ Clean reconnection logic (tear down old socket, spawn new one)
• ✅ Connection pooling for multiple concurrent traces
• ✅ Proper memory cleanup and garbage collection
• ✅ Retry mechanisms without listener duplication
• ✅ Long-term stability through modular, testable design

Made changes.